# _*_ coding: utf-8 _*_
"""
@ 时间    ：2024/10/23 13:06
@ 作者    ：旺财
@ 文件    ：02 K临近算法回归模型.py
@ 说明    ：k临近算法简单回归模型
"""

from sklearn.neighbors import KNeighborsRegressor
X = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]
y = [1, 2, 3, 4, 5]

model = KNeighborsRegressor(n_neighbors=2)
model.fit(X, y)

print(model.predict([[5, 5]]))